from flask import g,redirect,url_for
from functools import wraps

def login_required(func):
    @wraps(func)
    def wrapper(*args,**kwargs):
        if hasattr(g,'user'):
            return func(*args,**kwargs)
        else:
            return redirect(url_for('user.login'))
    return wrapper

def login_required_ajax(func):
    @wraps(func)
    def wrapper(*args,**kwargs):
        if hasattr(g,'user'):
            return func(*args,**kwargs)
        else:
            return {'success':False,'msg':'系统错误，请重新登陆'}
    return wrapper